from flask import Flask
from flask_socketio import SocketIO
from application.ws.test import init_test_websocket
from application.ws.plc import init_plc_websocket
def init_ws(app: Flask):
    # 创建一个主SocketIO实例
    main_socketio = SocketIO(app, cors_allowed_origins="*", async_mode="gevent")
    
    # 导入并初始化两个WebSocket服务，使用同一个SocketIO实例
   
    
    # 修改两个服务的初始化函数，使其接受SocketIO实例而不是app
    init_test_websocket(main_socketio)
    init_plc_websocket(main_socketio)
    
    # 将主SocketIO实例附加到app上
    app.socketio = main_socketio